/*
Copyright [2020] [https://www.xiaonuo.vip]

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

  http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：

1.请不要删除和修改根目录下的LICENSE文件。
2.请不要删除和修改Snowy源码头部的版权声明。
3.请保留源码和相关描述文件的项目出处，作者声明等。
4.分发源码时候，请注明软件出处 https://gitee.com/xiaonuobase/snowy
5.在修改包名，模块名称，项目代码等时，请注明软件出处 https://gitee.com/xiaonuobase/snowy
6.若您的项目无法满足以上几点，可申请商业授权，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
 */
package vip.xiaonuo.modular.tableinfo.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import vip.xiaonuo.core.consts.CommonConstant;
import vip.xiaonuo.core.enums.CommonStatusEnum;
import vip.xiaonuo.core.exception.ServiceException;
import vip.xiaonuo.core.factory.PageFactory;
import vip.xiaonuo.core.pojo.page.PageResult;
import vip.xiaonuo.core.util.PoiUtil;
import vip.xiaonuo.modular.databaseinfo.service.impl.BiDataSource;
import vip.xiaonuo.modular.tableinfo.entity.TableInfo;
import vip.xiaonuo.modular.tableinfo.enums.TableInfoExceptionEnum;
import vip.xiaonuo.modular.tableinfo.mapper.TableInfoMapper;
import vip.xiaonuo.modular.tableinfo.param.TableInfoParam;
import vip.xiaonuo.modular.tableinfo.service.TableInfoService;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Resource;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.List;
import java.util.Properties;

/**
 * 数据库信息表service接口实现类
 *
 * @author wfy
 * @date 2022-07-17 22:10:31
 */
@Service
public class TableInfoServiceImpl extends ServiceImpl<TableInfoMapper, TableInfo> implements TableInfoService {

    @Override
    public PageResult<TableInfo> page(TableInfoParam tableInfoParam) {
        QueryWrapper<TableInfo> queryWrapper = new QueryWrapper<>();
        if (ObjectUtil.isNotNull(tableInfoParam)) {

            // 根据数据库id 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getDatabaseId())) {
                queryWrapper.lambda().eq(TableInfo::getDatabaseId, tableInfoParam.getDatabaseId());
            }
            // 根据表权限分级 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTablePermissions())) {
                queryWrapper.lambda().eq(TableInfo::getTablePermissions, tableInfoParam.getTablePermissions());
            }
            // 根据表中文名称 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableCnName())) {
                queryWrapper.lambda().eq(TableInfo::getTableCnName, tableInfoParam.getTableCnName());
            }
            // 根据表用户 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableSchema())) {
                queryWrapper.lambda().eq(TableInfo::getTableSchema, tableInfoParam.getTableSchema());
            }
            // 根据表类型 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableType())) {
                queryWrapper.lambda().eq(TableInfo::getTableType, tableInfoParam.getTableType());
            }
            // 根据表英文名称 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableEnName())) {
                queryWrapper.lambda().eq(TableInfo::getTableEnName, tableInfoParam.getTableEnName());
            }
            // 根据该库内表的总数 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableSum())) {
                queryWrapper.lambda().eq(TableInfo::getTableSum, tableInfoParam.getTableSum());
            }
            // 根据该库内表内字段的总数 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableFieldSum())) {
                queryWrapper.lambda().eq(TableInfo::getTableFieldSum, tableInfoParam.getTableFieldSum());
            }
            // 根据表的创建时间 查询
            if (ObjectUtil.isNotEmpty(tableInfoParam.getTableCreateTime())) {
                queryWrapper.lambda().eq(TableInfo::getTableCreateTime, tableInfoParam.getTableCreateTime());
            }
        }
        return new PageResult<>(this.page(PageFactory.defaultPage(), queryWrapper));
    }

    @Override
    public List<TableInfo> list(TableInfoParam tableInfoParam) {
        return this.list();
    }

    @Override
    public void add(TableInfoParam tableInfoParam) {
        TableInfo tableInfo = new TableInfo();
        BeanUtil.copyProperties(tableInfoParam, tableInfo);
        this.save(tableInfo);
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void delete(List<TableInfoParam> tableInfoParamList) {
        tableInfoParamList.forEach(tableInfoParam -> {
            this.removeById(tableInfoParam.getId());
        });
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void edit(TableInfoParam tableInfoParam) {
        TableInfo tableInfo = this.queryTableInfo(tableInfoParam);
        BeanUtil.copyProperties(tableInfoParam, tableInfo);
        this.updateById(tableInfo);
    }

    @Override
    public TableInfo detail(TableInfoParam tableInfoParam) {
        return this.queryTableInfo(tableInfoParam);
    }

    /**
     * 获取数据库信息表
     *
     * @author wfy
     * @date 2022-07-17 22:10:31
     */
    private TableInfo queryTableInfo(TableInfoParam tableInfoParam) {
        TableInfo tableInfo = this.getById(tableInfoParam.getId());
        if (ObjectUtil.isNull(tableInfo)) {
            throw new ServiceException(TableInfoExceptionEnum.NOT_EXIST);
        }
        return tableInfo;
    }

    @Override
    public void export(TableInfoParam tableInfoParam) {
        List<TableInfo> list = this.list(tableInfoParam);
        PoiUtil.exportExcelWithStream("SnowyTableInfo.xls", TableInfo.class, list);
    }

    public static boolean setTaleRe(BiDataSource dataSource,String tableName,String tableRe) throws SQLException {
        Properties connectionProps = new Properties();
        connectionProps.put("user", dataSource.username);
        connectionProps.put("password", dataSource.password);
        connectionProps.put("remarks", "true");
        connectionProps.put("useInformationSchema", "true");
        try {
            Connection connection = DriverManager.getConnection(dataSource.url, connectionProps);
            Statement Statement = connection.createStatement();
            String setTableReSql = "alter table "+tableName+" comment '"+tableRe+"'";
            Statement.executeUpdate(setTableReSql);
        }catch (Exception e){
            System.err.println("[ERROR] TableInfoServiceImpl.java->setTaleRe : error = " +e.toString());
            return  false;
        }
        return true;
    }

}
